﻿using Microsoft.AspNetCore.Mvc.Controllers;
using Microsoft.AspNetCore.Mvc.Filters;
using Microsoft.AspNetCore.Mvc;
using AVCUMPlatform.HelperLibrary;
using AVCUMPlatform.Entities;
using HelperLibrary;
using Microsoft.Extensions.Caching.Memory;
using System.Collections;
using System.Reflection;
using System.Text.RegularExpressions;

namespace AVCUMPlatform
{
    /// <summary>
    /// 授权认证过滤器
    /// </summary>
    public class AuthorizationFilter : ActionFilterAttribute
    {
        private static readonly IMemoryCache _cache = new Microsoft.Extensions.Caching.Memory.MemoryCache(new MemoryCacheOptions());
        /// <summary>
        /// 执行前
        /// </summary>
        /// <param name="context"></param>
        public override void OnActionExecuting(ActionExecutingContext context)
        {
            // 添加接口调用日志
            bool isDefined = false;
            var controllerActionDescriptor = context.ActionDescriptor as ControllerActionDescriptor;
            if (controllerActionDescriptor != null)
            {
                isDefined = controllerActionDescriptor.MethodInfo.GetCustomAttributes(inherit: true)
                    .Any(a => a.GetType().Equals(typeof(NoNeedLogin)));
            }
            if (isDefined)
            {
                return;
            }
            //var content = JsonConvert.SerializeObject(context.ActionArguments);
            //string source = string.Empty;
            //if (context.ActionArguments.FirstOrDefault().Key == "messagequeues")
            //{
            //    var obj = context.ActionArguments.FirstOrDefault().Value;
            //    source = obj.GetType().GetProperty("Source").GetValue(obj, null).ToString();
            //}
            //if (source == "AVCBooking")
            //{
            //    string token = context.HttpContext.Request.Headers["token"];
            //    if (string.IsNullOrWhiteSpace(token))
            //    {
            //        context.Result = new UnauthorizedObjectResult(new
            //        {
            //            Code = 401,
            //            Msg = "你授权已过期，请重新授权"
            //        });
            //        Log.Write($"401:你授权已过期，请重新授权,token:{token}");
            //        return;
            //    }
            //    var res = AASData.VerifyToken(token);
            //    if (res.msg_id != 0)
            //    {
            //        context.Result = new UnauthorizedObjectResult(new
            //        {
            //            Code = 401,
            //            Msg = "你授权已过期，请重新授权"
            //        });
            //        Log.Write($"401:你授权已过期，请重新授权,token:{token},res:{JsonConvert.SerializeObject(res)}");
            //        return;
            //    }
            //}

            string token = context.HttpContext.Request.Headers["token"];
            Log.Write($"token:{token}");
            Log.Write($"GettokenExists:{Exists(token)}");
            Log.Write($"Gettoken:{Get<Source>(token)}");
            var source = Get<Source>(token);
            if (source == null)
            {
                context.Result = new UnauthorizedObjectResult(new
                {
                    Code = 401,
                    Msg = "你授权已过期，请重新授权"
                });
                return;
            }
            base.OnActionExecuting(context);
        }

        #region 缓存

        /// <summary>
        /// 验证缓存项是否存在
        /// </summary>
        /// <param name="key">缓存Key</param>
        /// <returns></returns>
        public static bool Exists(string key)
        {
            if (key == null)
                throw new ArgumentNullException(nameof(key));
            return _cache.TryGetValue(key, out _);
        }

        /// <summary>
        /// 添加缓存
        /// </summary>
        /// <param name="key">缓存Key</param>
        /// <param name="value">缓存Value</param>
        /// <param name="expiresSliding">滑动过期时长（如果在过期时间内有操作，则以当前时间点延长过期时间）</param>
        /// <param name="expiressAbsoulte">绝对过期时长</param>
        /// <returns></returns>
        public static bool Set(string key, object value, TimeSpan expiresSliding, TimeSpan expiressAbsoulte)
        {
            if (key == null)
                throw new ArgumentNullException(nameof(key));
            if (value == null)
                throw new ArgumentNullException(nameof(value));

            _cache.Set(key, value,
                new MemoryCacheEntryOptions().SetSlidingExpiration(expiresSliding)
                    .SetAbsoluteExpiration(expiressAbsoulte));
            return Exists(key);
        }

        /// <summary>
        /// 添加缓存
        /// </summary>
        /// <param name="key">缓存Key</param>
        /// <param name="value">缓存Value</param>
        /// <param name="expiresIn">缓存时长</param>
        /// <param name="isSliding">是否滑动过期（如果在过期时间内有操作，则以当前时间点延长过期时间）</param>
        /// <returns></returns>
        public static bool Set(string key, object value, TimeSpan expiresIn, bool isSliding = false)
        {
            if (key == null)
                throw new ArgumentNullException(nameof(key));
            if (value == null)
                throw new ArgumentNullException(nameof(value));

            _cache.Set(key, value,
                isSliding
                    ? new MemoryCacheEntryOptions().SetSlidingExpiration(expiresIn)
                    : new MemoryCacheEntryOptions().SetAbsoluteExpiration(expiresIn));

            return Exists(key);
        }

        /// <summary>
        /// 删除缓存
        /// </summary>
        /// <param name="key">缓存Key</param>
        /// <returns></returns>
        public static void Remove(string key)
        {
            if (key == null)
                throw new ArgumentNullException(nameof(key));

            _cache.Remove(key);
        }

        /// <summary>
        /// 批量删除缓存
        /// </summary>
        /// <returns></returns>
        public static void RemoveAll(IEnumerable<string> keys)
        {
            if (keys == null)
                throw new ArgumentNullException(nameof(keys));

            keys.ToList().ForEach(item => _cache.Remove(item));
        }

        /// <summary>
        /// 获取缓存
        /// </summary>
        /// <param name="key">缓存Key</param>
        /// <returns></returns>
        public static T Get<T>(string key)
        {
            if (key == null)
                return default(T);

            return (T)_cache.Get(key);
        }

        /// <summary>
        /// 获取缓存
        /// </summary>
        /// <param name="key">缓存Key</param>
        /// <returns></returns>
        public static object Get(string key)
        {
            if (key == null)
                return null;

            return _cache.Get(key);
        }

        /// <summary>
        /// 获取缓存集合
        /// </summary>
        /// <param name="keys">缓存Key集合</param>
        /// <returns></returns>
        public static IDictionary<string, object> GetAll(IEnumerable<string> keys)
        {
            if (keys == null)
                throw new ArgumentNullException(nameof(keys));

            var dict = new Dictionary<string, object>();
            keys.ToList().ForEach(item => dict.Add(item, _cache.Get(item)));
            return dict;
        }

        /// <summary>
        /// 删除所有缓存
        /// </summary>
        public static void RemoveCacheAll()
        {
            var l = GetCacheKeys();
            foreach (var s in l)
            {
                Remove(s);
            }
        }

        /// <summary>
        /// 删除匹配到的缓存
        /// </summary>
        /// <param name="pattern"></param>
        /// <returns></returns>
        public static void RemoveCacheRegex(string pattern)
        {
            IList<string> l = SearchCacheRegex(pattern);
            foreach (var s in l)
            {
                Remove(s);
            }
        }

        /// <summary>
        /// 搜索 匹配到的缓存
        /// </summary>
        /// <param name="pattern"></param>
        /// <returns></returns>
        public static IList<string> SearchCacheRegex(string pattern)
        {
            var cacheKeys = GetCacheKeys();
            var l = cacheKeys.Where(k => Regex.IsMatch(k, pattern)).ToList();
            return l.AsReadOnly();
        }

        /// <summary>
        /// 获取所有缓存键
        /// </summary>
        /// <returns></returns>
        public static List<string> GetCacheKeys()
        {
            const BindingFlags flags = BindingFlags.Instance | BindingFlags.NonPublic;
            var entries = _cache.GetType().GetField("_entries", flags).GetValue(_cache);
            var cacheItems = entries as IDictionary;
            var keys = new List<string>();
            if (cacheItems == null) return keys;
            foreach (DictionaryEntry cacheItem in cacheItems)
            {
                keys.Add(cacheItem.Key.ToString());
            }

            return keys;
        }

        #endregion
    }

    /// <summary>
    /// 免登录属性
    /// </summary>
    public class NoNeedLogin : ActionFilterAttribute
    {
        /// <summary>
        /// 执行前
        /// </summary>
        /// <param name="context"></param>
        public override void OnActionExecuting(ActionExecutingContext context)
        {
            if (context.ActionArguments.FirstOrDefault().Key != "login" && context.ActionArguments.FirstOrDefault().Key != "authentication")
            {
                if (CacheHelper.Get("login_token") == null)
                {
                    context.Result = new BadRequestObjectResult(new
                    {
                        Code = 401,
                        Msg = "你尚未登陆或授权已过期，请重新登陆"
                    });
                    return;
                }
            }

            base.OnActionExecuting(context);
        }
    }
}
